package com.cy.login.comment.util;

import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 * redis对字符串数据的通用操作类
 */
@Component
public class RedisMapper {

    @Resource
    private RedisTemplate<Object, Object> redisTemplate;

    /**
     * 根据键保存字符串
     *
     * @param key
     * @param val
     * @return
     */
    public boolean setKey(String key, String val) {
        redisTemplate.opsForValue().set(key, val);
        if (redisTemplate.hasKey(key)) {
            return true;
        } else {
            return false;
        }
    }

    /**
     * 根据键删除
     *
     * @param key
     * @return
     */
    public boolean delKey(String key) {
        if (redisTemplate.delete(key)) {
            return true;
        } else {
            return false;
        }
    }

    /**
     * 添加并设置过期时间
     *
     * @param key
     * @param val
     * @param timeout
     * @param unit
     * @return
     */
    public boolean setKey(String key, String val, long timeout, TimeUnit unit) {
        redisTemplate.opsForValue().set(key, val, timeout, unit);
        if (redisTemplate.hasKey(key)) {
            return true;
        } else {
            return false;
        }
    }

    /**
     * 根据键获得值
     *
     * @param key
     * @return
     */
    public Object getKey(String key) {
        if (redisTemplate.hasKey(key)) {
            Object value = redisTemplate.opsForValue().get(key);
            return value;
        } else {
            return null;
        }
    }

    /**
     * 保存hash数据到redis
     *
     * @param key1  大键
     * @param key2  小键
     * @param value 值，如果是对象使用json字符串
     * @return
     */
    public boolean setHash(String key1, String key2, String value) {
        try {
            redisTemplate.opsForHash().put(key1, key2, value);
            return true;
        } catch (Exception e) {
            return false;
        }
    }


    /**
     * 取出reids中的hash数据
     *
     * @param key1 大键
     * @param key2 小键
     * @return
     */
    public String getHash(String key1, String key2) {
        String result = null;
        try {
            result = redisTemplate.opsForHash().get(key1, key2).toString();
            return result;
        } catch (Exception e) {
            return null;
        }
    }

    /**
     * 根据大键取出hash中所有的数据
     *
     * @param key
     * @return
     */
    public Map<Object, Object> getHashEntries(String key) {
        Map<Object, Object> map = redisTemplate.opsForHash().entries(key);
        if (map != null) {
            return map;
        }
        return null;
    }

    /**
     * 判断指定的key是否存在
     *
     * @param key
     * @return
     */
    public Boolean hasKey(String key) {
        try {
            if (redisTemplate.hasKey(key)) {
                return true;
            }
        } catch (Exception e) {
            return false;
        }
        return false;
    }


    /**
     * 删除has中指定的键
     *
     * @param key1 大键
     *@param key2  小键
     * @return
     */
    public boolean delHasKey(String key1, String key2) {
        try {
            Long n = redisTemplate.opsForHash().delete(key1, key2);
            if (n > 0) {
                return true;
            }else {
                return false;
            }
        } catch (Exception e) {
            return false;
        }
    }


}
